import random
import string

class VerificationCodeGenerator:
    def __init__(self):
        # 定义所有可能的字符，包括大小写字母和数字
        self.characters = string.ascii_letters + string.digits

    def generate_code(self, length):
        """
        生成指定长度的验证码
        :param length: 验证码的长度
        :return: 生成的验证码
        """
        if length <= 0:
            raise ValueError("验证码长度必须为正整数")
        # 使用 random.choices 从所有可能的字符中随机选择指定长度的字符
        code = ''.join(random.choices(self.characters, k=length))
        return code
